python Train.py --gpus 0 --batch_size 5